Can you plot in Python